package com.jg.generator;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import com.baomidou.mybatisplus.generator.fill.Property;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * @author 86130
 * @version 1.0
 * @description: 代码生成器
 * @date 2025/9/16 10:42
 */
public class CodeGenerator {
    public static void main(String[] args) {
        // 使用 FastAutoGenerator 快速配置代码生成器
        //数据库配置
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/shopping_db",
                        "root", "root")
                //全局配置
                .globalConfig(builder -> {
                    builder.author("86130") // 设置作者
                            .dateType(DateType.ONLY_DATE)// 设置时间类型策略
                            .commentDate("yyyy-MM-dd HH:mm") // 设置注释日期格式
                            .enableSpringdoc()//doc文档
                            .outputDir(System.getProperty("user.dir")+ "/src/main/java") // 输出目录
                            .disableOpenDir(); // 生成后不打开资源管理器
                })
                //包配置
                .packageConfig(builder -> {
                    builder.parent("com.jg") // 设置父包名
                            .entity("model.entity") // 设置实体类包名
                            .mapper("mapper") // 设置 Mapper 接口包名
                            .service("service") // 设置 Service 接口包名
                            .serviceImpl("service.impl") // 设置 Service 实现类包名
                            .controller("controller") // 设置 Controller 包名
                            .xml("mapper"); // 设置 Mapper XML 文件包名
                    // 自定义 mapper.xml 的输出路径
                    builder.pathInfo(Collections.singletonMap(OutputFile.xml,
                            System.getProperty("user.dir") + "/src/main/resources/mapper"));
                })
                //策略配置
                .strategyConfig((scanner, builder) -> {
                    builder.addInclude(getTables(scanner.apply("请输入表名，多个英文逗号分隔？所有输入 all"))) // 设置需要生成的表名
                            .addTablePrefix("sys_")//设置过滤表前缀
                            //Entity 策略
                            .entityBuilder()
                            // .superClass(BaseEntity.class)//父类
                            // .disableSerialVersionUID()//禁用生成 serialVersionUID
                            .naming(NamingStrategy.underline_to_camel)//数据库表映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel)//下划线转驼峰命名
                            .addSuperEntityColumns("created_at", "updated_at", "deleted_at", "created_by","updated_by","version")//添加父类公共字段
                            .enableLombok() // 启用 Lombok
                            .enableTableFieldAnnotation() // 启用字段注解
                            // 添加表填充字段
                            .addTableFills(new Column("created_at", FieldFill.INSERT))
                            .addTableFills(new Property("updated_at", FieldFill.INSERT_UPDATE))
                            .idType(IdType.AUTO);
                    // Controller 策略
                    builder.controllerBuilder()
                            .enableRestStyle(); // 启用 REST 风格
                    // Service 策略
                    builder.serviceBuilder()
                            .formatServiceFileName("%sService") // 格式化 service 接口文件名称
                            .formatServiceImplFileName("%sServiceImpl"); // 格式化 service 实现类文件名称
                    // Mapper 策略
                    builder.mapperBuilder()
                            .enableMapperAnnotation() // 开启 @Mapper 注解
                            .enableBaseResultMap() // 启用 BaseResultMap 生成
                            .enableBaseColumnList() // 启用 BaseColumnList
                            .formatMapperFileName("%sMapper") // 格式化 mapper 文件名称
                            .formatXmlFileName("%sMapper"); // 格式化 xml 实现类文件名称
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用 Freemarker 模板引擎
                .execute(); // 执行生成

    }

    // 处理 all 情况
    protected static List<String> getTables(String tables) {
        return "all".equals(tables) ? Collections.emptyList() : Arrays.asList(tables.split(","));
    }


}
